iT邦幫忙

2024 iThome 鐵人賽

DAY 17
0
AI/ ML & Data

我的深度學習-從0開始實作物件偵測系列 第 17

【Day 17】實戰演練 - 問題與解決

  • 分享至 

  • xImage
  •  

問題描述與解決

由於是初次進行實作,在過程中遇到了滿多不同的問題,像一開始資料的分割以及處理等,最難受的就是套件版本的問題,當初在安裝套件時,參考了ultralytics官方的依賴項進行安裝套件,結果在訓練時,跳出了數多套件引發的錯誤,在一知半解的狀況下完全沒有頭緒,依賴項如下圖。
截圖 2024-07-29 上午11.53.02
錯誤的資訊上出現了Pytorch不可用的狀況,但使用了torch.cuda.is_available()得出的回應是True,這讓筆者感到很疑惑。

由於筆者對此也沒有頭緒,於是便試著尋找ChatGPT求助,將問題丟上去後,得出的回應卻是如下圖的流水文,即使嘗試了問題仍然依舊。
截圖 2024-07-29 下午1.48.36
後來借助著實驗室的學長幫忙,以及上官方的github上爬文是否有人有遇到相同的問題,終於是讓筆者爬到了這篇文章,問題如下圖,與筆者發生的狀況相同。
截圖 2024-07-29 下午1.52.03
後來有翻到官方人員對這個問題的回應,似乎是torchvision的版本上出現了問題導致CUDA的不支援,嘗試了更新版本後,果然錯誤都被解決了。
截圖 2024-07-29 下午1.53.12
這個問題困擾了筆者不少時間,因為一開始的問題錯誤並不是直接指向torch方面的問題,而是其他套件的錯誤,因此筆者當時只針對出現錯誤的項目去進行降版本以及爬文修正的部分,但問題仍然依舊,但後續通過這篇issue以及學長幫忙debug後,終於順利地解決。

得到解決後,模型也順利的訓練完了,後續筆者將對訓練完的模型做精準度以及對車牌實際的測試,那這邊我們下篇見。

image


上一篇
【Day 16】實戰演練 - 模型訓練 !
下一篇
【Day 18】實戰演練 - 模型評估!
系列文
我的深度學習-從0開始實作物件偵測30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言